from locust import HttpLocust, TaskSet, task

#定义用户行为
class UserBehavior(TaskSet):

    # 第一页的blog
    @task(5)
    def flasky_page_1(self):
        self.client.get("/api/v1.0/posts/")

    # 第一个帖子，第一页的评论
    @task(1)
    def flasky_comments_1(self):
        self.client.get("/api/v1.0/posts/1/comments/")


class WebsiteUser(HttpLocust):
    task_set = UserBehavior
    min_wait = 1000
    max_wait = 2000
